Update app.py
Browse files
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():
|