Sreekan commited on
Commit
c9d4799
·
verified ·
1 Parent(s): fae6640

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -339,10 +339,10 @@ from langchain_huggingface import HuggingFacePipeline
339
  from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
340
  import gradio as gr
341
 
342
- import torch
343
- print(torch.cuda.is_available()) # Check if a GPU is available
344
- a = torch.Tensor([1]).cuda() # Should succeed without errors
345
- print(a)
346
 
347
  # Define HuggingFace LLM
348
  def create_llm():
 
339
  from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
340
  import gradio as gr
341
 
342
+ #import torch
343
+ #print(torch.cuda.is_available()) # Check if a GPU is available
344
+ #a = torch.Tensor([1]).cuda() # Should succeed without errors
345
+ #print(a)
346
 
347
  # Define HuggingFace LLM
348
  def create_llm():